package com.plus.utils;

import cn.hutool.core.codec.Base64;
import cn.hutool.crypto.SecureUtil;
import cn.hutool.crypto.asymmetric.KeyType;
import cn.hutool.crypto.asymmetric.RSA;

import java.security.KeyPair;
import java.security.PrivateKey;
import java.security.PublicKey;

/**
 * @program: springboot-future
 * @ClassName RSAUtil
 * @description:AI
 * @author: 黄涛
 * @create: 2024-01-22 10:45
 * @Version 1.0
 **/
public class RSAUtil {

    public static final String ENCRYPT_TYPE = "RSA";

    public static void main(String[] args) {
        generateKeyPair();
    }

    /**
     * 自助生成密钥对
     */
    public static void generateKeyPair(){
        try {
            KeyPair pair = SecureUtil.generateKeyPair(ENCRYPT_TYPE);
            PrivateKey privateKey = pair.getPrivate();
            PublicKey publicKey = pair.getPublic();
            // 获取 公钥和私钥 的 编码格式（通过该 编码格式 可以反过来 生成公钥和私钥对象）
            byte[] pubEncBytes = publicKey.getEncoded();
            byte[] priEncBytes = privateKey.getEncoded();
            // 把公钥和私钥的编码格式转换为Base64文本
            String pubEncBase64 = Base64.encode(pubEncBytes);
            String priEncBase64 = Base64.encode(priEncBytes);
            System.out.println(String.format("私钥是：%s",priEncBase64));
            System.out.println(String.format("公钥是：%s",pubEncBase64));
        } catch (Exception e){
            e.printStackTrace();
        }
    }



    //region 公钥加密，私钥解密
    /**
     * 公钥加密
     * @param content 要加密的内容
     * @param publicKey 公钥
     */
    public static String encrypt(String content, String publicKey) {
        try{
            RSA rsa = new RSA(null,publicKey);
            return rsa.encryptBase64(content, KeyType.PublicKey);
        } catch (Exception e){
            e.printStackTrace();
        }
        return null;
    }

    /**
     * 私钥解密
     * @param content 要解密的内容
     * @param privateKey 私钥
     */
    public static String decrypt(String content, String privateKey) {
        try {
            RSA rsa = new RSA(privateKey,null);
            return rsa.decryptStr(content, KeyType.PrivateKey);
        } catch (Exception e){
            e.printStackTrace();
        }
        return null;
    }
    //endregion




}
